package com.axend.aerosense.common.connect.update.firmware;

import com.axend.aerosense.base.utils.ByteUtils;
import com.axend.aerosense.base.utils.Values;
import com.axend.aerosense.common.connect.tcp.RadarProvide;

import java.nio.ByteBuffer;
import java.util.ArrayList;
import java.util.List;

/**
 * 固件升级命令
 * @author ChenHongWei
 * @date :2020/12/1 16:20
 */

public class CommandList {

    /**
     * ********************************************
     *                                            *
     * 设备固件升级命令 1.0                         *
     *                                            *
     **********************************************/
    //外层命令头部
    public static final byte[] FIRMWARE_HEAD = {
            (byte) 0xAA,
            (byte) 0xBB,
            0x00
    };

    //外层命令尾部
    public static final byte[] FIRMWARE_END = {
            (byte) 0xCC,
            (byte) 0xDD
    };

    //开始固件升级命令(Break中断命令)
    public static final byte[] FIRMWARE_START_UPDATA = {
            (byte) 0x88,
            (byte) 0x99,
            (byte) 0xAA,
            (byte) 0xBB,
            (byte) 0xCC,
            (byte) 0xDD,
            (byte) 0xEE,
            (byte) 0xFF
    };

    //Erase擦除源文件
    public static final byte[] FIRMWARE_ERASE = {
            0x05,
            (byte) 0xAA,
            0x00,
            0x03,
            0x28,
            0x28,
            0x30,
            (byte) 0xC3,
            (byte) 0xCC,
            (byte) 0xDD
    };

    //设备回复的ACK
    public static final byte[] FIRMWARE_ACK = {
            0x00,
            0x04,
            (byte) 0xCC,
            0x00,
            (byte) 0xCC
    };

    //擦除命令
    public static byte[] getEraseCommand(){
        return ByteUtils.mergeByteArray(FIRMWARE_HEAD, FIRMWARE_ERASE);
    }

    //获取Open命令
    public static byte[] getOpenCommand(int fileSize){
        byte[] b0 = {0x15};
        byte[] b1 = {(byte) 0xAA, 0x00, 0x13};
        byte[] b2 = new byte[1];
        byte[] b3 = {0x21};
        byte[] b4 = ByteUtils.intToBytesLow2High(fileSize);
        byte[] b5 = {0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x00};
        b2[0] = checkSum(ByteUtils.mergeByteArray(b1, b2, b3, b4, b5));
        byte[] b8 = getCRC(ByteUtils.mergeByteArray(FIRMWARE_HEAD, b0, ByteUtils.mergeByteArray(b1, b2, b3, b4, b5)));
        return ByteUtils.mergeByteArray(FIRMWARE_HEAD, b0, b1, b2, b3, b4, b5, b8, FIRMWARE_END);
    }

    //发送文件命令
    public static byte[] sendFileCommand(byte[] file){
        //头
        byte[] b0 = {(byte) 0xAA};
        //length
        byte[] b1 = {0x00, ByteUtils.intToBytesLow2High(file.length + 3)[3]};
        //校验和
        byte[] b2 = new byte[1];
        //标识位
        byte[] b3 = {0x24};
        //计算校验和
        b2[0] = checkSum(ByteUtils.mergeByteArray(b0, b1, b2, b3, file));
        //内部命令
        byte[] playLoad = ByteUtils.mergeByteArray(b0, b1, b2, b3, file);
        byte[] playLoadLength = {ByteUtils.intToBytesLow2High(playLoad.length)[3]};
        byte[] crc = getCRC(ByteUtils.mergeByteArray(FIRMWARE_HEAD, playLoadLength, playLoad));
        return ByteUtils.mergeByteArray(FIRMWARE_HEAD, playLoadLength, playLoad, crc, FIRMWARE_END);
    }

    //关闭命令
    public static byte[] getCloseCommand(){
        //头部和标识符
        byte[] b0 = {(byte) 0xAA, 0x00, 0x07};
        //校验和
        byte[] sum = new byte[1];
        byte[] flag = {0x22};
        byte[] storageType = {0x00, 0x00, 0x00, 0x02};
        sum[0] = checkSum(ByteUtils.mergeByteArray(b0, sum, flag, storageType));
        //内部命令
        byte[] playload = ByteUtils.mergeByteArray(b0, sum, flag, storageType);

        byte[] playloadLength = {0x09};
        byte[] crc = getCRC(ByteUtils.mergeByteArray(FIRMWARE_HEAD, playloadLength, playload));
        return ByteUtils.mergeByteArray(FIRMWARE_HEAD, playloadLength, playload, crc, FIRMWARE_END);
    }

    //复位命令
    public static byte[] getResetCommand(){
        return new byte[]{
                (byte) 0xAA,
                (byte) 0xBB,
                0x02,
                0x00,
                0x50,
                (byte) 0xB9,
                (byte) 0xCC,
                (byte) 0xDD
        };
    }

    public static void printHexString(String str, byte[] bytes){
        //System.out.println(str + "------->" + ByteUtils.bytesToHexString(bytes));
    }



    /**
     * ********************************************
     *                                            *
     * 设备固件升级命令 2.0                         *
     *                                            *
     **********************************************/

    //获取开始升级命令
    public static byte[] getStartUpdateCommand(int size){
        byte[] data = ByteUtils.intToBytesLow2High(size);
        byte[] bytes = RadarProvide.mergeByte(ByteUtils.hexStringToBytes(Values.START_FIRMWARE_UPDATE), data);
        return bytes;
    }

    //获取封装发送的数据包
    public static byte[] getSendDataPack(byte[] data){
        ByteBuffer buffer = ByteBuffer.allocate(data.length + 2);
        buffer.put(data);
        buffer.put(getCRC(data));
        byte[] bytes = mergeByte(ByteUtils.hexStringToBytes(Values.SEND_FIRMWARE_DATA), buffer.array());
        return bytes;
    }

    //通知雷达固件下发完毕，升级完成
    public static byte[] getUpdateSuccessCommand(){
        return RadarProvide.mergeByte(ByteUtils.hexStringToBytes(Values.UPDATE_SUCCESS), ByteUtils.hexStringToBytes(Values.GET_DATA));
    }


    /**
     * 计算校验和
     * @param bytes
     * @return
     */
    private static byte checkSum(byte[] bytes){
        byte temp = 0x00;
        for (int i = 3; i < bytes.length; i++)
        {
            temp = (byte)((temp + bytes[i]) & 0xFF);
        }
        return temp;
    }

    /**
     * 查表法计算CRC16校验
     *
     * @param data 需要计算的字节数组
     */
    private static byte[] getCRC(byte[] data) {
        byte[] crc16_h = {
                (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40,
                (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41,
                (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41,
                (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40,
                (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41,
                (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40,
                (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40,
                (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41,
                (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41,
                (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40,
                (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40,
                (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41,
                (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40,
                (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41,
                (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41,
                (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x01, (byte) 0xC0, (byte) 0x80, (byte) 0x41, (byte) 0x00, (byte) 0xC1, (byte) 0x81, (byte) 0x40
        };

        byte[] crc16_l = {
                (byte) 0x00, (byte) 0xC0, (byte) 0xC1, (byte) 0x01, (byte) 0xC3, (byte) 0x03, (byte) 0x02, (byte) 0xC2, (byte) 0xC6, (byte) 0x06, (byte) 0x07, (byte) 0xC7, (byte) 0x05, (byte) 0xC5, (byte) 0xC4, (byte) 0x04,
                (byte) 0xCC, (byte) 0x0C, (byte) 0x0D, (byte) 0xCD, (byte) 0x0F, (byte) 0xCF, (byte) 0xCE, (byte) 0x0E, (byte) 0x0A, (byte) 0xCA, (byte) 0xCB, (byte) 0x0B, (byte) 0xC9, (byte) 0x09, (byte) 0x08, (byte) 0xC8,
                (byte) 0xD8, (byte) 0x18, (byte) 0x19, (byte) 0xD9, (byte) 0x1B, (byte) 0xDB, (byte) 0xDA, (byte) 0x1A, (byte) 0x1E, (byte) 0xDE, (byte) 0xDF, (byte) 0x1F, (byte) 0xDD, (byte) 0x1D, (byte) 0x1C, (byte) 0xDC,
                (byte) 0x14, (byte) 0xD4, (byte) 0xD5, (byte) 0x15, (byte) 0xD7, (byte) 0x17, (byte) 0x16, (byte) 0xD6, (byte) 0xD2, (byte) 0x12, (byte) 0x13, (byte) 0xD3, (byte) 0x11, (byte) 0xD1, (byte) 0xD0, (byte) 0x10,
                (byte) 0xF0, (byte) 0x30, (byte) 0x31, (byte) 0xF1, (byte) 0x33, (byte) 0xF3, (byte) 0xF2, (byte) 0x32, (byte) 0x36, (byte) 0xF6, (byte) 0xF7, (byte) 0x37, (byte) 0xF5, (byte) 0x35, (byte) 0x34, (byte) 0xF4,
                (byte) 0x3C, (byte) 0xFC, (byte) 0xFD, (byte) 0x3D, (byte) 0xFF, (byte) 0x3F, (byte) 0x3E, (byte) 0xFE, (byte) 0xFA, (byte) 0x3A, (byte) 0x3B, (byte) 0xFB, (byte) 0x39, (byte) 0xF9, (byte) 0xF8, (byte) 0x38,
                (byte) 0x28, (byte) 0xE8, (byte) 0xE9, (byte) 0x29, (byte) 0xEB, (byte) 0x2B, (byte) 0x2A, (byte) 0xEA, (byte) 0xEE, (byte) 0x2E, (byte) 0x2F, (byte) 0xEF, (byte) 0x2D, (byte) 0xED, (byte) 0xEC, (byte) 0x2C,
                (byte) 0xE4, (byte) 0x24, (byte) 0x25, (byte) 0xE5, (byte) 0x27, (byte) 0xE7, (byte) 0xE6, (byte) 0x26, (byte) 0x22, (byte) 0xE2, (byte) 0xE3, (byte) 0x23, (byte) 0xE1, (byte) 0x21, (byte) 0x20, (byte) 0xE0,
                (byte) 0xA0, (byte) 0x60, (byte) 0x61, (byte) 0xA1, (byte) 0x63, (byte) 0xA3, (byte) 0xA2, (byte) 0x62, (byte) 0x66, (byte) 0xA6, (byte) 0xA7, (byte) 0x67, (byte) 0xA5, (byte) 0x65, (byte) 0x64, (byte) 0xA4,
                (byte) 0x6C, (byte) 0xAC, (byte) 0xAD, (byte) 0x6D, (byte) 0xAF, (byte) 0x6F, (byte) 0x6E, (byte) 0xAE, (byte) 0xAA, (byte) 0x6A, (byte) 0x6B, (byte) 0xAB, (byte) 0x69, (byte) 0xA9, (byte) 0xA8, (byte) 0x68,
                (byte) 0x78, (byte) 0xB8, (byte) 0xB9, (byte) 0x79, (byte) 0xBB, (byte) 0x7B, (byte) 0x7A, (byte) 0xBA, (byte) 0xBE, (byte) 0x7E, (byte) 0x7F, (byte) 0xBF, (byte) 0x7D, (byte) 0xBD, (byte) 0xBC, (byte) 0x7C,
                (byte) 0xB4, (byte) 0x74, (byte) 0x75, (byte) 0xB5, (byte) 0x77, (byte) 0xB7, (byte) 0xB6, (byte) 0x76, (byte) 0x72, (byte) 0xB2, (byte) 0xB3, (byte) 0x73, (byte) 0xB1, (byte) 0x71, (byte) 0x70, (byte) 0xB0,
                (byte) 0x50, (byte) 0x90, (byte) 0x91, (byte) 0x51, (byte) 0x93, (byte) 0x53, (byte) 0x52, (byte) 0x92, (byte) 0x96, (byte) 0x56, (byte) 0x57, (byte) 0x97, (byte) 0x55, (byte) 0x95, (byte) 0x94, (byte) 0x54,
                (byte) 0x9C, (byte) 0x5C, (byte) 0x5D, (byte) 0x9D, (byte) 0x5F, (byte) 0x9F, (byte) 0x9E, (byte) 0x5E, (byte) 0x5A, (byte) 0x9A, (byte) 0x9B, (byte) 0x5B, (byte) 0x99, (byte) 0x59, (byte) 0x58, (byte) 0x98,
                (byte) 0x88, (byte) 0x48, (byte) 0x49, (byte) 0x89, (byte) 0x4B, (byte) 0x8B, (byte) 0x8A, (byte) 0x4A, (byte) 0x4E, (byte) 0x8E, (byte) 0x8F, (byte) 0x4F, (byte) 0x8D, (byte) 0x4D, (byte) 0x4C, (byte) 0x8C,
                (byte) 0x44, (byte) 0x84, (byte) 0x85, (byte) 0x45, (byte) 0x87, (byte) 0x47, (byte) 0x46, (byte) 0x86, (byte) 0x82, (byte) 0x42, (byte) 0x43, (byte) 0x83, (byte) 0x41, (byte) 0x81, (byte) 0x80, (byte) 0x40
        };

        int crc = 0x0000ffff;
        int ucCRCHi = 0x00ff;
        int ucCRCLo = 0x00ff;
        int iIndex;
        for (int i = 0; i < data.length; ++i) {
            iIndex = (ucCRCLo ^ data[i]) & 0x00ff;
            ucCRCLo = ucCRCHi ^ crc16_h[iIndex];
            ucCRCHi = crc16_l[iIndex];
        }

        crc = ((ucCRCHi & 0x00ff) << 8) | (ucCRCLo & 0x00ff) & 0xffff;
        //高低位互换，输出符合相关工具对Modbus CRC16的运算
        crc = ( (crc & 0xFF00) >> 8) | ( (crc & 0x00FF ) << 8);
        byte[] temp = ByteUtils.intToBytesLow2High(crc);
        return new byte[]{temp[2], temp[3]};
    }

    /**
     * 合并、转换命令
     * 此处合并方法只能用于合并发送固件内容命令。其他命令请勿使用此方法
     * @param tag
     * @param data
     * @return
     */
    public static byte[] mergeByte(byte[] tag, byte[] data){
        byte[] head = Values.HEAD_BYTE;
        byte[] lengthByte = ByteUtils.intToBytesHigh2Low(data.length);
        byte[] length = {lengthByte[0], lengthByte[1]};
        List<byte[]> list = new ArrayList<>();
        list.add(head);
        list.add(length);
        list.add(tag);
        list.add(data);
        byte[] result = ByteUtils.mergeByteArray(list);
        return result;
    }

}
